package com.lagou.job.config;

import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import redis.clients.jedis.HostAndPort;
import redis.clients.jedis.JedisCluster;
import redis.clients.jedis.JedisPoolConfig;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

/**
 * Jedis配置类
 * @author yzh
 */
@Configuration
public class JedisConfig {

    /**
     * 集群节点集合
     */
    @Value("#{'${spring.redis.cluster.nodes}'.split(',')}")
    private List<String> nodes;

    /**
     * 超时时间
     */
    @Value("${spring.redis.timeout}")
    private Integer timeout;

    /**
     * 连接池最大连接数（使用负值表示没有限制）
     */
    @Value("${spring.redis.jedis.pool.max-active}")
    private Integer maxTotal;

    /**
     * 连接池最大阻塞等待时间（使用负值表示没有限制）
     */
    @Value("${spring.redis.jedis.pool.max-wait}")
    private Integer maxWaitMillis;

    /**
     * 连接池中的最大空闲连接
     */
    @Value("${spring.redis.jedis.pool.max-idle}")
    private Integer maxIdle;

    /**
     * 连接池中的最小空闲连接
     */
    @Value("${spring.redis.jedis.pool.min-idle}")
    private Integer minIdle;

    /**
     * JedisCluster Bean
     * @return
     */
    @Bean
    public JedisCluster jedisCluster() {
        JedisPoolConfig jedisPoolConfig = new JedisPoolConfig();
        jedisPoolConfig.setMaxTotal(maxTotal);
        jedisPoolConfig.setMaxWaitMillis(maxWaitMillis);
        jedisPoolConfig.setMaxIdle(maxIdle);
        jedisPoolConfig.setMinIdle(minIdle);
        Set<HostAndPort> jedisClusterNode = new HashSet<>();
        for (int i = 0; i < nodes.size(); i++) {
            String node = nodes.get(i);
            int index = node.indexOf(":");
            jedisClusterNode.add(new HostAndPort(node.substring(0,index), Integer.parseInt(node.substring(index+1))));
        }
        JedisCluster jedisCluster = new JedisCluster(jedisClusterNode, timeout, jedisPoolConfig);
        return jedisCluster;
    }
}
